package com.ruoyi.material.service.impl;

import java.util.List;
import java.util.Objects;
import java.util.stream.Collectors;

import com.ruoyi.common.core.domain.model.TreeNode;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import com.ruoyi.material.mapper.MateriaTypeMapper;
import com.ruoyi.material.domain.MateriaType;
import com.ruoyi.material.service.IMateriaTypeService;

/**
 * 材料类型Service业务层处理
 * 
 * @author songxin
 * @date 2024-07-10
 */
@Service
public class MateriaTypeServiceImpl implements IMateriaTypeService 
{
    @Autowired
    private MateriaTypeMapper materiaTypeMapper;

    /**
     * 查询材料类型
     * 
     * @param materiatypeId 材料类型主键
     * @return 材料类型
     */
    @Override
    public MateriaType selectMateriaTypeByMateriatypeId(Long materiatypeId)
    {
        return materiaTypeMapper.selectMateriaTypeByMateriatypeId(materiatypeId);
    }

    /**
     * 查询材料类型列表
     * 
     * @param materiaType 材料类型
     * @return 材料类型
     */
    @Override
    public List<MateriaType> selectMateriaTypeList(MateriaType materiaType)
    {
        return materiaTypeMapper.selectMateriaTypeList(materiaType);
    }

    /**
     * 新增材料类型
     * 
     * @param materiaType 材料类型
     * @return 结果
     */
    @Override
    public int insertMateriaType(MateriaType materiaType)
    {
        return materiaTypeMapper.insertMateriaType(materiaType);
    }

    /**
     * 修改材料类型
     * 
     * @param materiaType 材料类型
     * @return 结果
     */
    @Override
    public int updateMateriaType(MateriaType materiaType)
    {
        return materiaTypeMapper.updateMateriaType(materiaType);
    }

    /**
     * 批量删除材料类型
     * 
     * @param materiatypeIds 需要删除的材料类型主键
     * @return 结果
     */
    @Override
    public int deleteMateriaTypeByMateriatypeIds(Long[] materiatypeIds)
    {
        return materiaTypeMapper.deleteMateriaTypeByMateriatypeIds(materiatypeIds);
    }

    /**
     * 删除材料类型信息
     * 
     * @param materiatypeId 材料类型主键
     * @return 结果
     */
    @Override
    public int deleteMateriaTypeByMateriatypeId(Long materiatypeId)
    {
        return materiaTypeMapper.deleteMateriaTypeByMateriatypeId(materiatypeId);
    }

    @Override
    public Long selectMateriaTypeName(MateriaType materiaType) {
        return materiaTypeMapper.selectMateriaTypeName(materiaType);
    }

    @Override
    public Long getIdByPid(Long materiatypeId) {
        return materiaTypeMapper.getIdByPid(materiatypeId);
    }

    @Override
    public Long getIdByConstraint(Long materiatypeId) {
        return materiaTypeMapper.getIdByConstraint(materiatypeId);
    }

    @Override
    public TreeNode getTree() {
        //查询数据库所有数据
        List<MateriaType> list = materiaTypeMapper.queryAll();
        //创建根节点
        TreeNode root = new TreeNode();
        root.setId(0L);
        root.setName("根节点");
        //给根节点添加子节点
        findchild(root,list);
        //返回根节点
        return root;
    }

    private void findchild(TreeNode root, List<MateriaType> list) {
        List<TreeNode> children = list.stream()
                //如果子节点的Pid和根节点的id相等，则找到子节点
                .filter(materiaType -> Objects.equals(root.getId(),materiaType.getMateriatypePid()))
                //将材料类型转成公共的树节点的类型
                .map(materiaType -> {
                    TreeNode node = new TreeNode();
                    node.setId(materiaType.getMateriatypeId());
                    node.setName(materiaType.getMateriatypeName());
                    node.setPid(materiaType.getMateriatypePid());
                    findchild(node,list);
                    return node;
                })
                .collect(Collectors.toList());
        //如果知道子节点。则将子节点集合放入上级的children中
        if(!children.isEmpty()){
            root.setChildren(children);
        }
    }


}
